import time
from tzer.tzer_seeds import my_model_seeds
from tvm import relax

MODEL_SEEDS = [
    my_model_seeds.get_model_0()[0],
    my_model_seeds.get_model_1()[0],
    my_model_seeds.get_model_2()[0],
    my_model_seeds.get_model_3()[0],
    my_model_seeds.get_model_4()[0],
    my_model_seeds.get_model_5()[0],
    my_model_seeds.get_model_6()[0],
    my_model_seeds.get_model_7()[0],
]

if __name__ == '__main__':
    for mod in MODEL_SEEDS:
        from tvm.contrib import coverage
        params = None
        before_cov = coverage.get_now()
        before_time = time.time()
        relax.build(mod, target='llvm', params=params)
        print(f'Increased cov: {coverage.get_now() - before_cov} in {time.time() - before_time} s')

    print(coverage.get_now())
